package info.wangyuan.agent.entity.po;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import info.wangyuan.agent.valid.ValidName;
import lombok.*;

import java.io.Serial;
import java.io.Serializable;
import java.util.List;

/**
 * 智能体配置实体类
 *
 * @author Albert
 * @since 2025-07-10 02:24:44
 */
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
@EqualsAndHashCode(callSuper = true)
@TableName(value = "tb_agents", autoResultMap = true)
public class AgentConfig extends BaseEntity implements Serializable {
    @Serial
    private static final long serialVersionUID = 1L;

    @TableId(type = IdType.AUTO)
    private Integer id;

    /**
     * 智能体名称
     */
    @ValidName
    private String name;

    /**
     * 绑定的系统提示词ID
     */
    private Integer systemPromptId;

    /**
     * 智能体欢迎语
     */
    private String welcomeMessage;

    /**
     * 聊天模型ID
     */
    private Integer chatModelId;

    /**
     * 启用状态
     */
    private Boolean enabled;

    /**
     * 备注
     */
    private String remark;

    /**
     * 知识库名称
     */
    @TableField(exist = false)
    private List<String> knowledgeBaseNames;
}
